package simple.tree;

import struct.TreeNode;

import java.util.ArrayList;
import java.util.List;

/**
 * @Desc <a href="https://leetcode.cn/problems/binary-tree-paths/">257. 二叉树的所有路径</a>
 * 给你一个二叉树的根节点 root ，按 任意顺序 ，返回所有从根节点到叶子节点的路径。
 * 叶子节点 是指没有子节点的节点。
 * 示例 1：
 *   输入：root = [1,2,3,null,5]
 *   输出：["1->2->5","1->3"]
 * 示例 2：
 *   输入：root = [1]
 *   输出：["1"]
 * 提示：
 *   树中节点的数目在范围 [1, 100] 内
 *   -100 <= Node.val <= 100
 * @Date 2023/2/6 1:50
 * @Created by 刘学松
 */
public class 二叉树的所有路径 {
    public List<String> binaryTreePaths(TreeNode root) {
        List<String> list = new ArrayList<>();
        search(new StringBuilder(), root, list);
        return list;
    }

    public void search(StringBuilder path, TreeNode node, List<String> list) {
        // 追加当前元素
        path.append(node.val);
        // 没有子节点，则直接添加到list
        if (node.left == null && node.right == null) {
            list.add(path.toString());
        } else {
            // 有子节点，则追加箭头符号
            path.append("->");
            // 有左子节点，则递归左子节点
            if (node.left != null) {
                search(new StringBuilder(path), node.left, list);
            }
            // 有右子节点，则递归右子节点
            if (node.right != null) {
                search(new StringBuilder(path), node.right, list);
            }
        }
    }
}
